package com.atguigu.gulimall.authserver.controller;

import com.alibaba.fastjson.TypeReference;
import com.atguigu.common.constant.AuthServerConstant;
import com.atguigu.common.exception.BizCodeEnum;
import com.atguigu.common.utils.R;
import com.atguigu.common.vo.MemberResponseVo;
import com.atguigu.gulimall.authserver.feign.MemberFeignService;
import com.atguigu.gulimall.authserver.feign.ThirdPartFeignService;
import com.atguigu.gulimall.authserver.vo.UserLoginVo;
import com.atguigu.gulimall.authserver.vo.UserRegisterVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.util.StringUtils;
import org.springframework.validation.BindingResult;
import org.springframework.validation.FieldError;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import javax.servlet.http.HttpSession;
import javax.validation.Valid;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

@Controller
public class LoginController {

    @Autowired
    private MemberFeignService memberFeignService;

    @Autowired
    private StringRedisTemplate redisTemplate;

    @Autowired
    private ThirdPartFeignService thirdPartFeignService;

    /**
     * 发送短信验证码给用户手机
     * 同时用户恶意刷验证码时返回错误信息。
     *
     * 验证码生成后将验证码拼接当前时间后保存在redis中，并且设置过期时间为10min
     * 如果用户在60秒以内，再次申请发送验证码，说明存在恶意刷验证码情况，这时返回错误信息。
     * @param phone 用户手机号
     */
    @GetMapping("/sms/sendCode")
    @ResponseBody
    public R sendCode(@RequestParam("phone") String phone) {
        // 1 todo: 1 接口防刷
        // 获取当前redis 中指定手机号是否有验证码
        // AuthServerConstant.SMS_CODE_CACHE_PREFIX 为自定义前缀，方便区分
        String redisCode = redisTemplate.opsForValue().get(AuthServerConstant.SMS_CODE_CACHE_PREFIX + phone);
        if (!StringUtils.isEmpty(redisCode)) {
            // redis中存在验证码，查看是否过了60秒
            long l = Long.parseLong(redisCode.split("_")[1]);
            if (System.currentTimeMillis() - l < 60000) {
                // 小于60秒，返回错误信息
                return R.error(BizCodeEnum.SMS_CODE_EXCEPTION.getCode(), BizCodeEnum.SMS_CODE_EXCEPTION.getMsg());
            }
        }

        // 2 再次验证
        String code = UUID.randomUUID().toString().substring(0, 5);
        // 使用当前时间拼接上验证码，用于二次发送时判断是否超过60秒
        String timeCode = code + "_" + System.currentTimeMillis();
        // 给redis 保存，指定保存时间为10分钟
        redisTemplate.opsForValue().set(AuthServerConstant.SMS_CODE_CACHE_PREFIX + phone, timeCode, 10, TimeUnit.MINUTES);
        // 交给第三方服务，发送验证码给手机，这里发送验证码需要申请，我申请失败，直接打印在控制台
        thirdPartFeignService.sendCode(phone, code);

        return R.ok();
    }

    /**
     * TODO: 重定向携带数据原理，利用session 携带数据将数据放在session 中
     * <p>
     * TODO: 1 分布式情况下的session 问题
     *
     * 用户注册：
     * 先验证用户数据合法性，再核对验证码，
     *
     * @param redirectAttributes 用于重定向携带数据，原生Model 在重定向时数据会丢失
     */
    @PostMapping("/register")
    public String register(@Valid UserRegisterVo vo, BindingResult result, RedirectAttributes redirectAttributes) {

        if (result.hasErrors()) {
            // 验证有错误
            List<FieldError> fieldErrors = new ArrayList<>(result.getFieldErrors());
            Map<String, String> errors = null;
            try {
                errors = fieldErrors.stream().collect(Collectors.toMap(FieldError::getField, FieldError::getDefaultMessage));
            } catch (Exception e) {
                e.printStackTrace();
            }
//            model.addAttribute("errors", errors);
            // 数据只能提取一次
            redirectAttributes.addFlashAttribute("errors", errors);

            return "redirect:http://auth.gulimall.com/reg.html";
        }

        // 注册逻辑
        // 1 校验验证码
        String code = vo.getCode();
        String phone = vo.getPhone();
        String redisCode = redisTemplate.opsForValue().get(AuthServerConstant.SMS_CODE_CACHE_PREFIX + phone);
        if (!StringUtils.isEmpty(redisCode)) {
            redisCode = redisCode.split("_")[0];
            if (redisCode.equalsIgnoreCase(code)) {
                // 删除验证码
                redisTemplate.delete(AuthServerConstant.SMS_CODE_CACHE_PREFIX + phone);
                // 验证码通过调用远程服务进行注册
                R r = memberFeignService.register(vo);
                if (r.getCode() == 0) {
                    // 成功
                    // 注册成功回到登录页
                    return "redirect:http://auth.gulimall.com/login.html";
                } else {
                    // 注册失败
                    Map<String, String> errors = new HashMap<>();
                    errors.put("msg", r.getData("msg", new TypeReference<String>(){}));
                    redirectAttributes.addFlashAttribute("errors", errors);

                    return "redirect:http://auth.gulimall.com/reg.html";
                }
            }
        }
        Map<String, String> errors = new HashMap<>();
        errors.put("code", "验证码错误");
        redirectAttributes.addFlashAttribute("errors", errors);

        return "redirect:http://auth.gulimall.com/reg.html";
    }

    /**
     * 登录
     * @param vo 用户登录信息
     * @param redirectAttributes 重定向时携带数据
     */
    @PostMapping("/login")
    public String login(UserLoginVo vo, RedirectAttributes redirectAttributes, HttpSession session){
        // 远程登录
        R r = memberFeignService.login(vo);
        if (r.getCode() == 0) {
            // 登录成功
            MemberResponseVo data = r.getData("data", new TypeReference<MemberResponseVo>() {
            });
            session.setAttribute(AuthServerConstant.LOGIN_USER, data);
            return "redirect:http://gulimall.com/index.html";
        } else {
            Map<String, String> errors = new HashMap<>();
            errors.put("msg", r.getData("msg", new TypeReference<String>(){}));
            redirectAttributes.addFlashAttribute("errors", errors);

            return "redirect:http://auth.gulimall.com/login.html";
        }
    }

    @GetMapping("/login.html")
    public String loginPage(HttpSession session) {
        Object attribute = session.getAttribute(AuthServerConstant.LOGIN_USER);
        if (attribute == null) {
            return "login";
        }

        return "redirect:http://gulimall.com";
    }
}
